#!/usr/bin/env python
# encoding: utf-8
'''
@author: Excelsiorly
@license: (C) Copyright 2022, All Rights Reserved.
@contact: excelsiorly@qq.com
@file: 008. 字符串转换整数 (atoi).py
@time: 2022/1/15 13:56
@desc: https://leetcode-cn.com/problems/string-to-integer-atoi/
> 请你来实现一个 myAtoi(string s) 函数，使其能将字符串转换成一个 32 位有符号整数（类似 C/C++ 中的 atoi 函数）。

函数 myAtoi(string s) 的算法如下：

    1. 读入字符串并丢弃无用的前导空格
    2. 检查下一个字符（假设还未到字符末尾）为正还是负号，读取该字符（如果有）。 确定最终结果是负数还是正数。 如果两者都不存在，则假定结果为正。
    3. 读入下一个字符，直到到达下一个非数字字符或到达输入的结尾。字符串的其余部分将被忽略。
    4. 将前面步骤读入的这些数字转换为整数（即，"123" -> 123， "0032" -> 32）。如果没有读入数字，则整数为 0 。必要时更改符号（从步骤 2 开始）。
    5. 如果整数数超过 32 位有符号整数范围 [−231,  231 − 1] ，需要截断这个整数，使其保持在这个范围内。具体来说，小于 −231 的整数应该被固定为 −231 ，大于 231 − 1 的整数应该被固定为 231 − 1 。
    6. 返回整数作为最终结果。

注意：
    - 本题中的空白字符只包括空格字符 ' ' 。
    - 除前导空格或数字后的其余字符串外，请勿忽略 任何其他字符。

@解题思路：
    1. 有限状态机，看题解吧。。：https://leetcode-cn.com/problems/string-to-integer-atoi/solution/zi-fu-chuan-zhuan-huan-zheng-shu-atoi-by-leetcode-/
    2. t: O(n), s: O(1)
'''
INT_MAX = 2 ** 31 - 1
INT_MIN = -2 ** 31


class Automaton():
    def __init__(self):
        self.ans = 0
        self.state = 'start'
        self.sign = 1
        self.table = {
            "start": ['start', 'signed', 'num', 'end'],
            'signed': ['end', 'end', 'num', 'end'],
            'num': ['end', 'end', 'num', 'end'],
            'end': ['end', 'end', 'end', 'end']
        }

    def get_col(self, c):
        if c == ' ': return 0
        if c == '+' or c == '-': return 1
        if '0' <= c <= '9': return 2
        return 3

    def get(self, c):
        self.state = self.table[self.state][self.get_col(c)]
        if self.state == 'num':
            self.ans = self.ans * 10 + int(c)
            self.ans = min(self.ans, INT_MAX) if self.sign == 1 else min(-INT_MIN, self.ans)
        if self.state == 'signed':
            self.sign = 1 if c == '+' else -1


class Solution(object):
    def myAtoi(self, s):
        """
        :type s: str
        :rtype: int
        """
        automaton = Automaton()
        for c in s:
            automaton.get(c)
        return automaton.ans * automaton.sign